//package com.liuyu.common.framework.generator.plus;
//
//import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
//import com.baomidou.mybatisplus.core.toolkit.StringPool;
//import com.baomidou.mybatisplus.generator.InjectionConfig;
//import com.baomidou.mybatisplus.generator.config.*;
//import com.baomidou.mybatisplus.generator.config.po.TableInfo;
//import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
//import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
//import org.apache.commons.lang3.StringUtils;
//
//import java.io.File;
//import java.nio.file.Paths;
//import java.util.ArrayList;
//import java.util.List;
//import java.util.Scanner;
//
///**
// * @ClassName CodeGenerator
// * @Author: liuyu
// * @Date: Created in 2020/4/4 19:13
// * @Version: 1
// * @Modified By:
// */
//// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
//public class MybatisPlusCodeGenerator {
//    private static String packageName = "com.asita.renovation";
//    private static String outDir = "F:\\java_web\\renovation\\src\\main\\java";
//    private static String entity = "entity";
//    private static String mapper = "mapper";
//    private static String service = "service";
//    private static String impl = "service.impl";
//    private static String controller = "controller";
//    private static String xml = "mapper.xml";
//    private static boolean isOverEntity = true;
//    private static boolean isOverController = false;
//    private static boolean isOverService = false;
//    private static boolean isOverServiceImpl = false;
//    private static boolean isOverMapper = false;
//    private static boolean isOverXml = false;
//
//    private static String entityVM = "/templates/entity.vm";
//    private static String controllerVM = "/templates/controller.vm";
//    private static String serviceVM = "";
//    private static String serviceImplVM = "";
//    private static String mapperVM = "";
//    private static String xmlVM = "";
//
//    private static String[] baseDir = {entity, mapper, service, impl, controller};
//
//    /**
//     * <p>
//     * 读取控制台内容
//     * </p>
//     */
//    public static String scanner(String tip) {
//        Scanner scanner = new Scanner(System.in);
//        StringBuilder help = new StringBuilder();
//        help.append("请输入" + tip + "：");
//        System.out.println(help.toString());
//        if (scanner.hasNext()) {
//            String ipt = scanner.next();
//            if (StringUtils.isNotEmpty(ipt)) {
//                return ipt;
//            }
//        }
//        throw new MybatisPlusException("请输入正确的" + tip + "！");
//    }
//
//    public static void main(String[] args) {
//        // 代码生成器
////        AutoGenerator mpg = new AutoGenerator();
//        //解决注释
//        MyAutoGenerator mpg = new MyAutoGenerator();
//        String module = "/iinsure-dao";
//        // 全局配置
//        GlobalConfig gc = new GlobalConfig();
//        String projectPath = System.getProperty("user.dir");
//        gc.setOutputDir(projectPath + module + "/src/main/java");
//        gc.setAuthor("liuyu");
//        gc.setOpen(false);
//
//        //实体属性 Swagger2 注解
//        gc.setSwagger2(true);
//        gc.setFileOverride(true);
//        gc.setEnableCache(false);
////        xml
//        gc.setBaseResultMap(true);
////        xml
//        gc.setBaseColumnList(true);
//        mpg.setGlobalConfig(gc);
//
//
//        // 数据源配置
//        DataSourceConfig dsc = new DataSourceConfig();
//        dsc.setUrl("jdbc:mysql://56.101.1.33:3306/iinsure?serverTimezone=UTC&useSSL=false&useUnicode=true&characterEncoding=UTF-8");
//        // dsc.setSchemaName("public");
//        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
//        dsc.setUsername("root");
//        dsc.setPassword("piccweb");
//        mpg.setDataSource(dsc);
//
//        // 包配置
//        PackageConfig pc = new PackageConfig();
//
//        pc.setModuleName(scanner("模块名"));
//        pc.setParent("com.liuyu.common.framework.mybatis");
//        mpg.setPackageInfo(pc);
//
//        // 自定义配置
//        InjectionConfig cfg = new InjectionConfig() {
//            @Override
//            public void initMap() {
//                // to do nothing
//            }
//        };
//
//        // 如果模板引擎是 freemarker
//        String templatePath = "/templates/mapper.xml.ftl";
//        // 如果模板引擎是 velocity
//        // String templatePath = "/templates/mapper.xml.vm";
//
//        // 自定义输出配置
//        List<FileOutConfig> focList = new ArrayList<>();
//        // 自定义配置会被优先输出
//        focList.add(new FileOutConfig(templatePath) {
//            @Override
//            public String outputFile(TableInfo tableInfo) {
//                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
//                return projectPath + module + "/src/main/resources/mapper/" + pc.getModuleName() + "/base"
//                        + "/" + tableInfo.getEntityName() + "MapperBase" + StringPool.DOT_XML;
//            }
//        });
//        String customerTemplatePath = "/templates/customMapper.xml.ftl";
//        focList.add(new FileOutConfig(customerTemplatePath) {
//            @Override
//            public String outputFile(TableInfo tableInfo) {
//                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
//                return projectPath + module + "/src/main/resources/mapper/" + pc.getModuleName() + "/custom"
//                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
//            }
//        });
//       /* cfg.setFileCreate(new IFileCreate() {
//            @Override
//            public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
//                String path = projectPath + module + "/src/main/resources/mapper/" + pc.getModuleName() + "/custom";
////                String gcpath = gc.getOutputDir()+;
//
//                checkDir(filePath);
//                if (filePath.contains(path)) {
//                    return false;
//                }
//
//                return false;
//            }
//        });*/
//        /*
//        cfg.setFileCreate(new IFileCreate() {
//            @Override
//            public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
//                // 判断自定义文件夹是否需要创建
//                checkDir("调用默认方法创建的目录");
//                return false;
//            }
//        });
//        */
//        cfg.setFileOutConfigList(focList);
//        mpg.setCfg(cfg);
//
//        // 配置模板
//        TemplateConfig templateConfig = new TemplateConfig();
//
//        // 配置自定义输出模板
//        //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
//        // templateConfig.setEntity("templates/entity2.java");
//        // templateConfig.setService();
//        // templateConfig.setController();
//
//        templateConfig.setXml(null);
//        mpg.setTemplate(templateConfig);
//
//
//        // 策略配置
//        StrategyConfig strategy = new StrategyConfig();
//        strategy.setNaming(NamingStrategy.underline_to_camel);
//        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
////        strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
//        strategy.setEntityLombokModel(true);
//        strategy.setRestControllerStyle(false);
//
//        // 公共父类
////        strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
//        // 写于父类中的公共字段
////        strategy.setSuperEntityColumns("id");
//        strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));
//        strategy.setControllerMappingHyphenStyle(true);
//        strategy.setTablePrefix(pc.getModuleName() + "_");
//
//        mpg.setStrategy(strategy);
//        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
//        mpg.execute();
//    }
//
//    /**
//     * 根据自己的需要，修改哪些包下面的 要覆盖还是不覆盖
//     * @param packageName
//     */
//    private static TemplateConfig initTemplateConfig(String packageName) {
//        TemplateConfig tc = new TemplateConfig();
//        for (String tmp : baseDir) {
//            initVM(tc);
//            File file = new File(Paths.get(outDir, String.join("/", packageName.split("\\.")), tmp).toString());
//            String[] list = file.list();
//            if (list != null && list.length > 0) {
//                if (!isOverController) {
//                    tc.setController(null);
//                }
//                if (!isOverService) {
//                    tc.setService(null);
//                }
//                if (!isOverServiceImpl) {
//                    tc.setServiceImpl(null);
//                }
//                if (!isOverEntity) {
//                    tc.setEntity(null);
//                }
//                if (!isOverMapper) {
//                    tc.setEntity(null);
//                }
//                if (!isOverXml) {
//                    tc.setXml(null);
//                }
//            }
//        }
//        return tc;
//    }
//
//
//    private static void initVM(TemplateConfig tc) {
//        if (stringIsNotNull(entityVM)) {
//            tc.setEntity(entityVM);
//        }
//        if (stringIsNotNull(mapperVM)) {
//            tc.setMapper(mapperVM);
//        }
//        if (stringIsNotNull(serviceImplVM)) {
//            tc.setServiceImpl(serviceImplVM);
//        }
//        if (stringIsNotNull(serviceVM)) {
//            tc.setService(serviceVM);
//        }
//        if (stringIsNotNull(xmlVM)) {
//            tc.setXml(xmlVM);
//        }
//        if (stringIsNotNull(controllerVM)) {
//            tc.setController(controllerVM);
//        }
//    }
//
//    /**
//     * 简单判断字符串是不是为空
//     * @param s
//     * @return
//     */
//    private static boolean stringIsNotNull(String s) {
//        if (null != s && !s.equals("") && s.length() > 0 && s.trim().length() > 0) {
//            return true;
//        }
//        return false;
//    }
//
//    /**
//     * 初始化包目录配置
//     * @return
//     */
//    private static PackageConfig initPackage() {
//        PackageConfig packageConfig = new PackageConfig();
//        packageConfig.setParent(packageName);
//        packageConfig.setService(service);
//        packageConfig.setServiceImpl(impl);
//        packageConfig.setController(controller);
//        packageConfig.setEntity(entity);
//        packageConfig.setXml(xml);
//        return packageConfig;
//    }
//
//
//}
